package com.frejoys.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.frejoys.common.dto.admin.UserLevelStoreMoneyLogListDto;
import com.frejoys.common.util.AppConfigUtil;
import com.frejoys.common.util.TimeUtil;
import com.frejoys.common.vo.admin.UserLevelStoreMoneyListVo;
import com.frejoys.common.vo.admin.UserLevelStoreMoneyLogListVo;
import com.frejoys.dao.entity.*;
import com.frejoys.dao.mapper.StoreMapper;
import com.frejoys.dao.mapper.UserLevelStoreMoneyLogMapper;
import com.frejoys.dao.mapper.UserMapper;
import com.frejoys.service.UserLevelStoreMoneyLogService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Service
@RequiredArgsConstructor
public class UserLevelStoreMoneyLogServiceImpl extends ServiceImpl<UserLevelStoreMoneyLogMapper, UserLevelStoreMoneyLog> implements UserLevelStoreMoneyLogService {

    private final UserMapper userMapper;

    private final StoreMapper storeMapper;

    @Override
    public boolean addInfo(Long userId, Integer storeId, Long drawSn, BigDecimal money, Integer type) {
        return save(new UserLevelStoreMoneyLog(){{
            setUserId(userId);
            setStoreId(storeId);
            setDrawSn(drawSn);
            setMoney(money);
            setCreateTime(TimeUtil.unixTime());
            setType(type);
        }});
    }


    @Override
    public IPage<UserLevelStoreMoneyLogListVo> userLevelStoreMoneyLogListInfo(UserLevelStoreMoneyLogListDto dto) {
        Page page = new Page<>(dto.getPage(), dto.getSize());

        LambdaQueryWrapper<UserLevelStoreMoneyLog> wrapper = new LambdaQueryWrapper<>();

        if (dto.getUserId() != null){
            wrapper.eq(UserLevelStoreMoneyLog::getUserId,dto.getUserId());
        }
        if (StrUtil.isNotEmpty(dto.getPhone())){
            Long userId = userMapper.phoneByUserId(dto.getPhone());

            if (userId == null){
                return new Page<>();
            }
            wrapper.eq(UserLevelStoreMoneyLog::getUserId,userId);
        }
        if (StrUtil.isNotEmpty(dto.getNickname())){
            List<Long> userIds = userMapper.nicknameByUserId(dto.getNickname());

            if (userIds.isEmpty()){
                return new Page<>();
            }
            wrapper.in(UserLevelStoreMoneyLog::getUserId,userIds);
        }
        if (StrUtil.isNotEmpty(dto.getStoreName())){
            List<Store> stores = storeMapper.getStoreNameById(dto.getStoreName());

            if (stores.isEmpty()){
                return new Page<>();
            }

            List<Integer> storeIds = stores.stream().map(Store::getId).toList();
            wrapper.in(UserLevelStoreMoneyLog::getStoreId,storeIds);
        }

        if (dto.getType() != null){
            wrapper.eq(UserLevelStoreMoneyLog::getType,dto.getType());
        }
        wrapper.orderByDesc(UserLevelStoreMoneyLog::getCreateTime);

        Page<UserLevelStoreMoneyLog> userLevelStoreMoneyLogPage = baseMapper.selectPage(page, wrapper);
        IPage<UserLevelStoreMoneyLogListVo> voIPage = userLevelStoreMoneyLogPage.convert(userLevelStoreMoney -> BeanUtil.copyProperties(userLevelStoreMoney, UserLevelStoreMoneyLogListVo.class));
        List<UserLevelStoreMoneyLogListVo> list = voIPage.getRecords();

        if (!list.isEmpty()) {
            // 推荐人id
            List<Long> userIds = list.stream().map(UserLevelStoreMoneyLogListVo::getUserId).distinct().toList();

            List<User> users = userMapper.getUserByIds(userIds);
            Map<Long, User> userMap = users.stream().collect(Collectors.toMap(User::getId, user -> user));

            // 店铺id
            List<Integer> storeIds = list.stream().map(UserLevelStoreMoneyLogListVo::getStoreId).toList();

            List<Store> stores = storeMapper.getStoreNameList(storeIds);
            Map<Integer, Store> storeMap = stores.stream().collect(Collectors.toMap(Store::getId, store -> store));

            list.forEach(vo ->{

                if (userMap.containsKey(vo.getUserId())){
                    User user = userMap.get(vo.getUserId());
                    vo.setNickname(user.getNickname());
                    vo.setAvatar(AppConfigUtil.getUrlPrefix(user.getAvatar()));
                    vo.setPhone(user.getPhone());
                }
                if (storeMap.containsKey(vo.getStoreId())){
                    Store store = storeMap.get(vo.getStoreId());
                    vo.setStoreName(store.getName());
                    vo.setLogoUrl(AppConfigUtil.getUrlPrefix(store.getLogo()));
                }
            });
        }
        return voIPage;
    }

}
